#!/bin/bash

cd ~/onnx_samples/onnx_transformer
python tf_cann_generator.py

# >>> conda initialize >>>
# !! Contents within this block are managed by 'conda init' !!
__conda_setup="$('/home/ubuntu/anaconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
if [ $? -eq 0 ]; then
    eval "$__conda_setup"
else
    if [ -f "/home/ubuntu/anaconda3/etc/profile.d/conda.sh" ]; then
        . "/home/ubuntu/anaconda3/etc/profile.d/conda.sh"
    else
        export PATH="/home/ubuntu/anaconda3/bin:$PATH"
    fi
fi
unset __conda_setup
# <<< conda initialize <<<

conda activate cann_5.1
#cd ~/Ascend/samples/cplusplus/level1_single_api/3_ir/IRBuild/out
. /home/ubuntu/Ascend/ascend-toolkit/set_env.sh
#./ir_build Ascend310 tf ${1}
atc --model=${1} --framework=3 --output=${2} --soc_version=Ascend310
